CUDA_VISIBLE_DEVICES=4 python test.py -m test